import torch
from torch import nn


class module_jackoak(nn.Module):

    def __init__(self) -> None:
        super().__init__()

    def forward(self, input):
        output = input + 1
        return output


def main():
    m_jackoak = module_jackoak()
    input = torch.tensor([1, 2, 3])

    output = m_jackoak(input)

    print(output)


if __name__ == '__main__':
    main()
